Remove Duplicates from Sorted Array
题目大意
对排好序的list去重,输出去重后长度,并且不能创建新的数组
解题思路
快慢指针
代码
官方答案
数组完成排序后,我们可以放置两个指针 ii 和 jj,其中 ii 是慢指针,而 jj 是快指针。只要 nums[i] = nums[j]nums[i]=nums[j],我们就增加 jj 以跳过重复项。
当我们遇到 nums[j] \neq nums[i]nums[j]≠nums[i] 时,跳过重复项的运行已经结束,因此我们必须把它(nums[j]nums[j])的值复制到 nums[i + 1]nums[i+1]。然后递增 ii,接着我们将再次重复相同的过程,直到 jj 到达数组的末尾为止。
1 2 3 4 5 6 7 8 9 10 11
| public int removeDuplicates(int[] nums) { if (nums.length == 0) return 0; int i = 0; for (int j = 1; j < nums.length; j++) { if (nums[j] != nums[i]) { i++; nums[i] = nums[j]; } } return i + 1; }
|
我提交的(不忍直视)
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
| class Solution(object): def removeDuplicates(self, nums): """ :type nums: List[int] :rtype: int """ j = 0 for i in range(len(nums)): if nums[i] == nums[j]: i += 1 continue else: nums[j+1] = nums[i] i += 1 j += 1 nums = nums[:j+1] return len(nums)
|
Remove Duplicates from Sorted Array II
题目大意
在 Remove Duplicates from Sorted Array(从一个有序的数组中去除重复的数字,返回处理后的数组长度) 的基础上,可以使每个数字最多重复一次,也就是说如果某一个数字的个数大于等于2个,结果中应保留2个该数字。
解题思路
参考:http://www.cnblogs.com/zuoyuan/p/3783453.html
使用两个指针prev和curr,判断A[curr]是否和A[prev]、A[prev-1]相等,如果相等curr指针继续向后遍历,直到不相等时,将curr指针指向的值赋值给A[prev+1]。
代码
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
| class Solution(object): def removeDuplicates(self, nums): """ :type nums: List[int] :rtype: int """ if len(nums) <= 2: return len(nums) prev = 1 curr = 2 while curr < len(nums): # print 'curr:', nums[curr], 'prev, prev-1:', nums[prev], nums[prev - 1] if nums[curr] == nums[prev] and nums[curr] == nums[prev - 1]: curr += 1 else: prev += 1 # print 'fu zhi gei prev hou de yi wei:', nums[prev], nums[curr] nums[prev] = nums[curr] curr += 1 return prev + 1
|
总结